tiny refactor
fixed regression