minor fix to reduce complexity