associate() not working on embedsOne

[mikebronner]

When trying the following:

$user->address()->associate($address);

I get the following error:

BadMethodCallException
Call to undefined method Jenssegers\Mongodb\Query\Builder::associate()

I use associate to add sub-documents to a model without having to make a database call using save(). After everything is associated to the main model, I then call save() on that.

This works for other relationships, could this be added to work with embedsOne?

Thanks! :)