Fix small bug (missing this.)