ClassCastException fixed

Integer instead of Long