Step 3 - a little bug fix