Second commit. Small bug fix.