Fix broken unit testing when typing 'rake test'