correcting code formatting for test