Refactored a test and added gem versions