Fixed missing test