Fixing damage computation logic