Minor change - remove unnecessary part of code