Adjust test

Indentation is incorrect, but this will pass for now.