Update infrastructure.storage implementation