Fixed methods in test need to add to main project