Fixed lint warnings in test