Fix failing tests
Merge pull request #4 from teddyzeenny/factory_injections

Fix failing tests