Removed warnings from test class