Small bug, updated