improved tests for the first step