Remove unused method
Make an integration test