Simple modification

Please review...