Fixed integration tests for repository
Added exception handler for all classes