Fixed typo for epochs in test file